package com.ls.utils;

import com.ls.domain.Student;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;

public class StudentExcelUtils {
    //总行数
    private static int totalRows = 100;
    //总列数
    private static int totalCells = 10;
    //错误信息接收器
    private static String errorMsg;
    /**
     * 验证EXCEL文件
     *
     * @param filePath
     * @return
     */
    public static boolean validateExcel(String filePath) {
        if (filePath == null || !(isExcel2003(filePath) || isExcel2007(filePath))) {
            errorMsg = "文件名不是excel格式";
            return false;
        }
        return true;
    }

    /*
        excel2003版本的excel文件是.xls格式，excel2007及以上版本的excel的是.xlsx格式。
    */
    // @描述：是否是2003的excel，返回true是2003
    public static boolean isExcel2003(String filePath)  {
        return filePath.matches("^.+\\.(?i)(xls)$");
    }
    //@描述：是否是2007的excel，返回true是2007
    public static boolean isExcel2007(String filePath)  {
        return filePath.matches("^.+\\.(?i)(xlsx)$");

    }

    /**
     * 读取Excel里面客户的信息
     * @param wb
     * @return
     */
    private static List<Student> readExcelValue(Workbook wb) {
        //默认会跳过第一行标题
        // 得到第一个shell
        Sheet sheet = wb.getSheetAt(0);
        // 得到Excel的行数
        totalRows = sheet.getPhysicalNumberOfRows();
        // 得到Excel的列数(前提是有行数)
        if (totalRows > 1 && sheet.getRow(0) != null) {
            totalCells = sheet.getRow(0).getPhysicalNumberOfCells();
        }
        List<Student> studentList = new ArrayList<Student>();
        // 循环Excel行数
        for (int r = 1; r < totalRows; r++) {
            Row row = sheet.getRow(r);
            if (row == null) {
                continue;
            }
            Student student = new Student();
            // 循环Excel的列
            for (int c = 0; c < totalCells ; c++) {
                Cell cell = row.getCell(c);
                if (null != cell) {
                    if (c == 0) {            //第一列
                        if (cell.getCellTypeEnum() == CellType.STRING) {
                            String stringCellValue = cell.getStringCellValue();
                            student.setSno(stringCellValue);
                        }
                    } else if (c == 1) {
                        if (cell.getCellTypeEnum() == CellType.STRING) {
                            String stringCellValue = cell.getStringCellValue();
                            student.setStudentName(stringCellValue);
                        }
                    } else if (c == 2) {
                        if (cell.getCellTypeEnum() == CellType.STRING) {
                            String stringCellValue = cell.getStringCellValue();
                            student.setStudentClass(stringCellValue);
                        }
                    } else if (c == 3) {
                        if (cell.getCellTypeEnum() == CellType.STRING) {
                            String stringCellValue = cell.getStringCellValue();
                            student.setStudentSex(stringCellValue);
                        }

                    }
                }
            }
            //将excel解析出来的数据赋值给对象添加到list中
            studentList.add(student);
        }
        return studentList;
    }

    /**
     * 根据excel里面的内容读取客户信息
     * @param is 输入流
     * @param isExcel2003 excel是2003还是2007版本
     * @return
     * @throws IOException
     */
    public static List<Student> createExcel(InputStream is, boolean isExcel2003) {
        try{
            Workbook wb = null;
            if (isExcel2003) {
                // 当excel是2003时,创建excel2003
                wb = new HSSFWorkbook(is);
            } else {
                // 当excel是2007时,创建excel2007
                wb = new XSSFWorkbook(is);
            }
            // 读取Excel里面客户的信息
            List<Student> studentList = readExcelValue(wb);
            return studentList;
        } catch (IOException e) {
            e.printStackTrace();
        }
        return null;
    }


    /**
     * 读EXCEL文件，获取信息集合
     * @return
     */
    public static List<Student> getExcelInfo(MultipartFile file) {
        String files = file.getOriginalFilename();//获取文件名
        try {
            if (!validateExcel(files)) {// 验证文件名是否合格
                return null;
            }
            boolean isExcel2003 = true;// 根据文件名判断文件是2003版本还是2007版本
            if (isExcel2007(files)) {
                isExcel2003 = false;
            }
            List<Student> studentList = createExcel(file.getInputStream(), isExcel2003);
            return studentList;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }

}
